package cn.ddm.hrm.service.impl;

import cn.ddm.hrm.domain.CourseType;
import cn.ddm.hrm.exception.ErrorCode;
import cn.ddm.hrm.exception.ValidUtils;
import cn.ddm.hrm.mapper.CourseTypeMapper;
import cn.ddm.hrm.service.ICourseTypeService;
import cn.ddm.hrm.vo.CrumbsVo;
import com.baomidou.mybatisplus.mapper.EntityWrapper;
import com.baomidou.mybatisplus.service.impl.ServiceImpl;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.cache.annotation.Cacheable;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;

import static sun.security.krb5.Confounder.longValue;

/**
 * <p>
 * 课程目录 服务实现类
 * </p>
 *
 * @author ldm
 * @since 2022-01-03
 */
@Service
public class CourseTypeServiceImpl extends ServiceImpl<CourseTypeMapper, CourseType> implements ICourseTypeService {

    @Autowired
    private RedisTemplate redisTemplate;

    private static final String COURSE_CACHE = "COURSE_CACHE";

    @Autowired
    private CourseTypeMapper courseTypeMapper;

    @Override
    public List<CourseType> treeData(long pid) {
        // 1.递归思想
//        return loadTreeData(pid);

        List<CourseType> courseTypes = (List<CourseType>) redisTemplate.opsForValue().get(COURSE_CACHE);

        if (courseTypes != null) {
            return courseTypes;
        } else {
            List<CourseType> courseTypeList = loadTreeDataByLoop(pid);
            redisTemplate.opsForValue().set(COURSE_CACHE, courseTypeList);
            return courseTypeList;
        }

        // 2.循环

        // 3.SQL递归
//        return courseTypeMapper.loadTreeDataBySql(pid);
    }

    @Override
    public List<CrumbsVo> crumbs(Long typeId) {
        // 检验参数
        ValidUtils.assertNotNull(typeId, ErrorCode.CODE_400_COURSE_TYPE_ILLEGAL);
        // 返回的list
        List<CrumbsVo> list = new ArrayList<>();
        // 查询当前自己
        String path = courseTypeMapper.selectById(typeId).getPath();
        // 对path进行分割
        String[] pathArr = path.split("\\.");
        // 遍历
        for (String id : pathArr) {
            CrumbsVo crumbsVo = new CrumbsVo();
            CourseType courseType = courseTypeMapper.selectById(id);
            crumbsVo.setOwnerCourseType(courseType);
            // 得到其他
            List<CourseType> courseTypes = courseTypeMapper.selectList(new EntityWrapper<CourseType>().eq("pid", courseType.getPid()));
            //除掉自己
            Iterator<CourseType> iterator = courseTypes.iterator();
            while (iterator.hasNext()) {
                CourseType type = iterator.next();
                if (courseType.getId().longValue() == type.getId().longValue()) {
                    iterator.remove();
                    break;
                }
            }
            crumbsVo.setOtherCourseTypes(courseTypes);
            list.add(crumbsVo);
        }
        // 返回list
        return list;
    }

    /**
     * 循环
     * @param pid
     * @return
     */
    private List<CourseType> loadTreeDataByLoop(Long pid) {
        // 1.结果集
        List<CourseType> result = new ArrayList<>();

        // 2.查询所有节点
        List<CourseType> courseTypes = courseTypeMapper.selectList(null);

        // 用一个map装进
        HashMap<Long, CourseType> map = new HashMap<>();
        for (CourseType courseType : courseTypes) {
            map.put(courseType.getId(), courseType);
        }
        // 遍历,判断是否为当前节点
        courseTypes.forEach(courseType -> {
            Long pid1 = courseType.getPid();
            if (pid.longValue() == pid1.longValue())
                result.add(courseType);
            else {
                // 获取父亲
                CourseType parent  = map.get(pid1);
                // 放进去
                parent.getChildren().add(courseType);
            }
        });

        // 返回结果
        return result;

    }

    /**
     * 递归查询
     * @param pid
     * @return
     */
    private List<CourseType> loadTreeData(long pid) {
        List<CourseType> courseTypes = courseTypeMapper.selectList(new EntityWrapper<CourseType>().eq("pid", pid));
        // 出口

        if (courseTypes == null)
            return null;

        courseTypes.forEach(courseType -> {
            Long id = courseType.getId();
            List<CourseType> courseTypeList = loadTreeData(id);
            courseType.setChildren(courseTypeList);
        });

        return courseTypes;

    }

    @Override
    public boolean insert(CourseType entity) {
        redisTemplate.delete(COURSE_CACHE);
        return super.insert(entity);
    }

    @Override
    public boolean deleteById(Serializable id) {
        redisTemplate.delete(COURSE_CACHE);
        return super.deleteById(id);
    }

    @Override
    public boolean updateById(CourseType entity) {
        redisTemplate.delete(COURSE_CACHE);
        return super.updateById(entity);
    }
}
